package com.bckj.fastboot.paid.biz.infrastructure.sdk;

import lombok.RequiredArgsConstructor;
import org.springframework.stereotype.Component;

import java.util.HashMap;
import java.util.Map;

@Component
@RequiredArgsConstructor
public class PayServiceManager {

    private final PayServiceWrapperFactory payServiceWrapperFactory;

    private final Map<String, PayServiceWrapper> payServiceMap = new HashMap<>();

    public PayServiceWrapper getPayService(String appId, String payWay) {
        PayServiceWrapper r = payServiceMap.get(appId + payWay);
        if (r != null) {
            return r;
        }
        PayServiceWrapper payServiceWrapper = payServiceWrapperFactory.getPayServiceWrapper(appId, payWay);
        if (payServiceWrapper == null) {
            return null;
        }
        setPayServiceWrapper(appId, payWay, payServiceWrapper);
        return payServiceWrapper;
    }

    public void setPayServiceWrapper(String appId, String payWay, PayServiceWrapper payServiceWrapper) {
        payServiceMap.put(appId + payWay, payServiceWrapper);
    }
}
